#!/usr/bin/env python
# -*- coding: utf-8 -*-
# @Time    : 2018/5/6 10:17
# @Author  : zx
# @Site    : 
# @File    : manage.py
# @Software: PyCharm_

from flask_script import Manager, Server
from app import app
from flaskdemo import db
from flaskdemo.models.model import User, Category

manage = Manager(app)
manage.add_command("server", Server())

@manage.shell
def make_shell_context():
    return dict(app=app, db=db, User=User, Category=Category)

if __name__ == '__main__':
    manage.run()
